export NCCL_P2P_LEVEL=NVL
export NCCL_P2P_DISABLE=1
accelerate launch --config_file ./config/accelerate_config.yaml train_vq.py